package com.mj.listen3.链表;

import java.util.Stack;

/**
 * https://leetcode-cn.com/problems/remove-linked-list-elements/
 */
public class _203_移除链表元素 {
    public ListNode removeElements(ListNode head, int val) {
        if (head == null) return head;
        // 建立一个虚拟头结点
        ListNode newHead = new ListNode(0);
        ListNode newTail = newHead;
        while (head != null) {
            if (head.val != val) {
                // 不相等，建立新的链表
                newTail.next = head;
                // 移动尾巴
                newTail = head;
            }
            // 相等的话就直接跳过
            head = head.next;
        }
        // 最后一个节点是删除的节点的话，newTail指向有问题，应该指向next
        newTail.next = null;
        // newHead最初是虚拟头节点
        return newHead.next;
    }

    public ListNode removeElements1(ListNode head, int val) {

        while (head != null && head.val == val) {
            // 从头开始一直都是val，就直接从头往后删除即可
            head = head.next;
        }

        if (head == null) return null;

        // 走这里说明头不是val
        ListNode pre = head;
        // 头不是val，从第二个节点开始
        ListNode cur = head.next;
        while (cur != null) {
            if (cur.val == val) {
                // 当前节点是要删除节点
                pre.next = cur.next;
            } else {
                // 记录上一个节点。cur.val == val的时候不移动pre，因为可能下一个节点还是要删除的节点
                pre = cur;
            }

            // 指针移动，指向下一个节点
            cur = cur.next;
        }
        return head;
    }

    public ListNode removeElements2(ListNode head, int val) {
        if (head == null) {
            return head;
        }

        Stack<ListNode> stack = new Stack<>();
        while (head != null) {
            if (head.val != val) {
                // 不为val的节点入栈
                stack.push(head);
            }
            head = head.next;
        }

        // 此时head为null

        // 出栈建立新的节点
        while (!stack.isEmpty()) {
            // 栈顶的节点指向尾节点
            stack.peek().next = head;
            // 弹出的节点为新的节点
            head = stack.pop();
        }
        return head;
    }
}